const mongoose = require('mongoose');
const Role = require('../models/roles');
require('dotenv').config();

async function initRoles() {
    try {
        // 连接数据库
        await mongoose.connect(process.env.MONGODB_URI);

        // 创建基本角色
        const roles = [
            {
                name: 'admin',
                role_desc: '系统管理员'
            },
            {
                name: 'user',
                role_desc: '普通用户'
            }
        ];

        // 插入角色
        for (const role of roles) {
            await Role.findOneAndUpdate(
                { name: role.name },
                role,
                { upsert: true, new: true }
            );
        }

        console.log('角色初始化成功');
        process.exit(0);
    } catch (error) {
        console.error('角色初始化失败:', error);
        process.exit(1);
    }
}

initRoles(); 